// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "ui/views/controls/native/native_view_host_test_base.h"

#include "base/macros.h"
#include "ui/views/controls/native/native_view_host.h"
#include "ui/views/widget/widget.h"

namespace views {
namespace test {

    // Testing wrapper of the NativeViewHost.
    class NativeViewHostTestBase::NativeViewHostTesting : public NativeViewHost {
    public:
        explicit NativeViewHostTesting(NativeViewHostTestBase* owner)
            : owner_(owner)
        {
        }
        ~NativeViewHostTesting() override { owner_->host_destroyed_count_++; }

    private:
        NativeViewHostTestBase* owner_;

        DISALLOW_COPY_AND_ASSIGN(NativeViewHostTesting);
    };

    NativeViewHostTestBase::NativeViewHostTestBase()
        : host_destroyed_count_(0)
    {
    }

    NativeViewHostTestBase::~NativeViewHostTestBase()
    {
    }

    void NativeViewHostTestBase::TearDown()
    {
        DestroyTopLevel();
        ViewsTestBase::TearDown();
    }

    void NativeViewHostTestBase::CreateTopLevel()
    {
        toplevel_.reset(new Widget);
        Widget::InitParams toplevel_params = CreateParams(Widget::InitParams::TYPE_WINDOW);
        toplevel_params.ownership = Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET;
        toplevel_->Init(toplevel_params);
    }

    void NativeViewHostTestBase::CreateTestingHost()
    {
        host_.reset(new NativeViewHostTesting(this));
    }

    Widget* NativeViewHostTestBase::CreateChildForHost(
        gfx::NativeView native_parent_view,
        View* parent_view,
        View* contents_view,
        NativeViewHost* host)
    {
        Widget* child = new Widget;
        Widget::InitParams child_params(Widget::InitParams::TYPE_CONTROL);
        child_params.ownership = Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET;
        child_params.parent = native_parent_view;
        child->Init(child_params);
        child->SetContentsView(contents_view);

        // Owned by |parent_view|.
        parent_view->AddChildView(host);
        host->Attach(child->GetNativeView());

        return child;
    }

    void NativeViewHostTestBase::DestroyTopLevel()
    {
        toplevel_.reset();
    }

    void NativeViewHostTestBase::DestroyHost()
    {
        host_.reset();
    }

    NativeViewHost* NativeViewHostTestBase::ReleaseHost()
    {
        return host_.release();
    }

    NativeViewHostWrapper* NativeViewHostTestBase::GetNativeWrapper()
    {
        return host_->native_wrapper_.get();
    }

} // namespace test
} // namespace views
